'''
PT Practice #3
Variation of Game Spinner 
Student selector - part 3
'''
from codex import *
import random
import time

peanuts = ["Snoopy", "Woodstock", "Charlie", "Linus", "Sally", 
           "Lucy", "Franklin", "Pigpen", "Violet"]

simpsons = ["Bart", "Homer", "Marge", "Lisa", "Maggie", 
            "Mr Burns", "Ned", "Apu", "Millhouse"]

def pick_student(period, count):
    delay = 0.0
    if period == 1:
        the_list = peanuts
        class_name = "Peanuts"
    else:
        the_list = simpsons
        class_name = "Simpsons"

    for number in range(count):
        student = random.choice(the_list)
        display.clear()
        display.print(class_name, scale=4)
        display.print(student, color=YELLOW, scale=4)
        delay = delay + 0.01
        time.sleep(delay)

# == Main Program ==
while True:
    if buttons.was_pressed(BTN_A):
        count = random.randrange(10, 20)
        pick_student(1, count)
    
    if buttons.was_pressed(BTN_B):
        count = random.randrange(10, 20)
        pick_student(2, count)

    if buttons.was_pressed(BTN_D):
        display.clear()
        display.print("Done")
        break